#!/bin/bash

# 检查是否提供了任意参数
if [ -n "$1" ]; then
    DELETE_ON_SUCCESS=true
else
    DELETE_ON_SUCCESS=false
fi

# 如果提供了任意参数，则删除 dky_onnx_tj 文件夹及其内容并克隆新的仓库
if [ "$DELETE_ON_SUCCESS" == true ]; then
    rm -rf dky_onnx_tj
    echo "dky_onnx_tj 文件夹及其内容已删除。"
    git clone https://gitee.com/xu_panpan/dky_onnx_tj.git
else
    echo "不删除 dky_onnx_tj 文件夹及其内容，也不克隆新的仓库。"
fi

# 进入 dky_onnx_tj 文件夹并创建 build 文件夹
if [ -d "dky_onnx_tj" ]; then
    cd dky_onnx_tj
    mkdir -p build
    cd build
    mkdir -p result_images
    mkdir -p result_labels 
	
    # 复制 yolov5 文件夹到当前目录
    cp -r /yolov5 .

    # 使用 CMake 配置项目
    cmake ..

    # 编译项目
    make

    # 运行示例程序
    ./opencv_example

    # 复制结果文件夹到指定位置
    cp -r ./result_images /yolov5/result_images
    cp -r ./result_labels /yolov5/result_labels
else
    echo "dky_onnx_tj 文件夹不存在，跳过编译和运行步骤。"
fi